/*
    * Copyright (C) 2010 The Guava Authors
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    * http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  package com.google.common.collect.testing;
  
  import com.google.common.annotations.GwtIncompatible;
  import com.google.common.collect.ImmutableSortedMap;
  import com.google.common.collect.Lists;
  import com.google.common.collect.Ordering;
  import com.google.common.collect.Sets;
  import com.google.common.collect.testing.features.CollectionFeature;
  import com.google.common.collect.testing.features.CollectionSize;
  import com.google.common.testing.SerializableTester;
  
  import junit.framework.Test;
  import junit.framework.TestCase;
  import junit.framework.TestSuite;
  
  import java.util.Arrays;
  import java.util.Collections;
  import java.util.List;
  import java.util.Map;
  import java.util.NavigableSet;
  import java.util.Set;
  import java.util.SortedSet;
  
  public class SafeTreeSetTest extends TestCase {
    public static Test suite() {
      TestSuite suite = new TestSuite();
      suite.addTestSuite(SafeTreeSetTest.class);
      suite.addTest(
          NavigableSetTestSuiteBuilder.using(new TestStringSetGenerator() {
            @Override protected Set<String> create(String[] elements) {
              return new SafeTreeSet<String>(Arrays.asList(elements));
            }
  
            @Override public List<String> order(List<String> insertionOrder) {
              return Lists.newArrayList(Sets.newTreeSet(insertionOrder));
            }
          }).withFeatures(CollectionSize.ANY, CollectionFeature.KNOWN_ORDER,
              CollectionFeature.GENERAL_PURPOSE).named(
              "SafeTreeSet with natural comparator").createTestSuite());
      suite.addTest(SetTestSuiteBuilder.using(new TestStringSetGenerator() {
        @Override protected Set<String> create(String[] elements) {
          NavigableSet<String> set =
              new SafeTreeSet<String>(Ordering.natural().nullsFirst());
          Collections.addAll(set, elements);
          return set;
        }
  
        @Override public List<String> order(List<String> insertionOrder) {
          return Lists.newArrayList(Sets.newTreeSet(insertionOrder));
        }
      }).withFeatures(CollectionSize.ANY, CollectionFeature.KNOWN_ORDER,
          CollectionFeature.GENERAL_PURPOSE, CollectionFeature.ALLOWS_NULL_VALUES)
          .named("SafeTreeSet with null-friendly comparator").createTestSuite());
      return suite;
    }
  
    @GwtIncompatible("SerializableTester")
    public void testViewSerialization() {
      Map<String, Integer> map =
          ImmutableSortedMap.of("one", 1, "two", 2, "three", 3);
      SerializableTester.reserializeAndAssert(map.entrySet());
      SerializableTester.reserializeAndAssert(map.keySet());
      assertEquals(Lists.newArrayList(map.values()),
          Lists.newArrayList(SerializableTester.reserialize(map.values())));
    }
  
    @GwtIncompatible("SerializableTester")
    public void testEmpty_serialization() {
      SortedSet<String> set = new SafeTreeSet<String>();
      SortedSet<String> copy = SerializableTester.reserializeAndAssert(set);
      assertEquals(set.comparator(), copy.comparator());
    }
  
    @GwtIncompatible("SerializableTester")
    public void testSingle_serialization() {
      SortedSet<String> set = new SafeTreeSet<String>();
      set.add("e");
      SortedSet<String> copy = SerializableTester.reserializeAndAssert(set);
      assertEquals(set.comparator(), copy.comparator());
    }
  
    @GwtIncompatible("SerializableTester")
    public void testSeveral_serialization() {
     SortedSet<String> set = new SafeTreeSet<String>();
     set.add("a");
     set.add("b");
     set.add("c");
     SortedSet<String> copy = SerializableTester.reserializeAndAssert(set);
     assertEquals(set.comparator(), copy.comparator());
   }
 }